Skip to content

Add config file for fused MoE for Nemotron (TP4, B200)#34411

Merged
vllm-bot merged 1 commit intovllm-project:mainfrom
de-inf:tune_moe
Feb 12, 2026
Merged

Add config file for fused MoE for Nemotron (TP4, B200)#34411
vllm-bot merged 1 commit intovllm-project:mainfrom
de-inf:tune_moe

Conversation

@danisereb
Copy link
Copy Markdown
Contributor

@danisereb danisereb commented Feb 12, 2026

Purpose

Tune MoE config for Nemotron-H model on B200 with TP4 (using benchmark_moe script).

Related PRs:
#32804
#33510

Test Plan

Compare performance with/without the JSON file using vllm bench serve.

Test Result

ISL 1024, OSL 1024

Batch size Output tok/s Output tok/s with JSON Perf gain
64 1449.72 1647.53 13.65%
256 2954.71 3361.69 13.77%
512 3635.13 4295.97 18.18%

Essential Elements of an Effective PR Description Checklist
  • The purpose of the PR, such as "Fix some issue (link existing issues this PR will resolve)".
  • The test plan, such as providing test command.
  • The test results, such as pasting the results comparison before and after, or e2e results
  • (Optional) The necessary documentation update, such as updating supported_models.md and examples for a new model.
  • (Optional) Release notes update. If your change is user facing, please update the release notes draft in the Google Doc.

Copy link
Copy Markdown
Contributor

@gemini-code-assist gemini-code-assist Bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code Review

This pull request introduces a new configuration file with tuned parameters for the fused Mixture of Experts (MoE) kernel. The configuration is specifically for the Nemotron-H model on NVIDIA B200 GPUs with a tensor parallelism of 4. The change is straightforward, adding a data file for performance optimization. The file format and naming convention are consistent with the existing structure. The changes look good and I don't see any issues.

Signed-off-by: Daniel Serebrenik <daserebrenik@nvidia.com>
@danisereb danisereb marked this pull request as ready for review February 12, 2026 09:54
@mgoin mgoin enabled auto-merge (squash) February 12, 2026 12:51
@mgoin
Copy link
Copy Markdown
Member

mgoin commented Feb 12, 2026

Thanks!

@github-actions github-actions Bot added the ready ONLY add when PR is ready to merge/full CI is needed label Feb 12, 2026
@vllm-bot vllm-bot merged commit dea6351 into vllm-project:main Feb 12, 2026
54 of 60 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ready ONLY add when PR is ready to merge/full CI is needed

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants